import unittest

from login_function import login


class Test117(unittest.TestCase):
    def test_method1(self):
        self.assertEqual('登录成功', login('admin', '123456'))

    def test_method2(self):
        self.assertEqual('用户名或密码错误，登录失败', login('admin', '123123'))

    def test_method3(self):
        self.assertEqual('用户名或密码错误，登录失败', login('admin', ''))

    def test_method4(self):
        self.assertIn('失败', login('aaa', '123456'))

